package net.lmc.realtime.jtp.dws.log.job;

import org.apache.flink.configuration.Configuration;
import org.apache.flink.table.api.EnvironmentSettings;
import org.apache.flink.table.api.Table;
import org.apache.flink.table.api.TableEnvironment;

public class JtpTrafficAppStartMinuteWindowDwsJob {
    public static void main(String[] args) {
        //执行环境
        TableEnvironment tabEnv = getTableEnv();

        //输入表 映射到kafka消息队列
        createInputTable(tabEnv);

        //数据处理
        Table reportTable = handle1(tabEnv);


        //输出表 映射到clickhouse
        createOutputTable(tabEnv);

        Table joinTable = joinDim(tabEnv);

        //保存数据到clickhouse
        saveToClickhouse(tabEnv,reportTable,joinTable);


    }

    private static Table joinDim(TableEnvironment tabEnv) {
        Table startAdReportTable = tabEnv.sqlQuery("SELECT" +
                " window_start AS window_start_time,\n" +
                " window_end AS window_end_time,\n" +
                " open_ad_id,\n" +
                " count(open_ad_id) AS open_ad_count,\n" +
                " sum(open_ad_ms) AS open_ad_ms,\n" +
                " sum(open_ad_skip_ms) AS open_ad_skip_ms,\n" +
                " UNIX_TIMESTAMP() * 1000 AS ts,\n" +
                " PROCTIME() AS proc_time\n" +
                " FROM TABLE (\n" +
                " TUMBLE(TABLE start_log_table,DESCRIPTOR(row_time),INTERVAL '10' second)\n" +
                " )t1\n" +
                " GROUP BY\n" +
                " window_start,window_end,open_ad_id"
        );

        tabEnv.createTemporaryView("start_ad_result_table2", startAdReportTable);

        tabEnv.executeSql("CREATE TABLE `ads_info_mysql_source_dim`\n" +
                "(\n" +
                "  id               INT COMMENT '主键ID',\n" +
                "  name             STRING COMMENT '广告名称',\n" +
                "  content          STRING COMMENT '广告内容（图片URL/视频URL/H5代码）',\n" +
                "  type             STRING COMMENT '广告类型（IMAGE, VIDEO, HTML5）',\n" +
                "  start_time       TIMESTAMP COMMENT '展示开始时间',\n" +
                "  end_time         TIMESTAMP COMMENT '展示结束时间',\n" +
                "  is_active        INT COMMENT '是否启用',\n" +
                "  target_condition STRING COMMENT '定向条件（JSON格式）',\n" +
                "  created_at       TIMESTAMP COMMENT '数据插入时间',\n" +
                "  updated_at       TIMESTAMP COMMENT '数据更新时间',\n" +
                "  PRIMARY KEY (id) NOT ENFORCED\n" +
                ") WITH (\n" +
                "  'connector' = 'jdbc',\n" +
                "  'url' = 'jdbc:mysql://node103:3306/jtp_ads',\n" +
                "  'table-name' = 'ads_info',\n" +
                "  'lookup.cache.max-rows' = '10',\n" +
                "  'lookup.cache.ttl' = '1 hour',\n" +
                "  'username' = 'root',\n" +
                "  'password' = '123456',\n" +
                "  'driver' = 'com.mysql.cj.jdbc.Driver'\n" +
                ")");

//        tabEnv.executeSql("select * from ads_info_mysql_source_dim").print();

        Table joinTable = tabEnv.sqlQuery("SELECT\n" +
                " t1.window_start_time,\n" +
                " t1.window_end_time,\n" +
                " t2.name AS open_ad_id,\n" +
                " t1.open_ad_count,\n" +
                " t1.open_ad_ms,\n" +
                " t1.open_ad_skip_ms,\n" +
                " t1.ts\n" +
                " FROM start_ad_result_table2 t1\n" +
                " LEFT JOIN ads_info_mysql_source_dim FOR SYSTEM_TIME AS OF t1.proc_time AS t2\n" +
                " ON CAST(t1.open_ad_id AS INT) = t2.id");

        return joinTable;

    }

    private static void saveToClickhouse(TableEnvironment tabEnv, Table startReportTable, Table joinTable) {

        tabEnv.createTemporaryView("start_report_table_click",startReportTable);
        //保存app启动类型汇总表
        tabEnv.executeSql("INSERT INTO dws_traffic_app_start_window_report_clickhouse_sink SELECT     " +
                "DATE_FORMAT(window_start_time, 'yyyy-MM-dd HH:mm:ss') AS window_start_time     " +
                ", DATE_FORMAT(window_end_time, 'yyyy-MM-dd HH:mm:ss') AS window_end_time     " +
                ", entry     " +
                ", entry_count     " +
                ", loading_time     " +
                ", ts FROM start_report_table_click"
        );

        //tabEnv.createTemporaryView("start_ad_report_table_click", startReportAdTable);
        tabEnv.createTemporaryView("join_table", joinTable);
        //保存app启动广告汇总表
        tabEnv.executeSql("INSERT INTO dws_traffic_app_start_ad_window_report_clickhouse_sink SELECT     " +
                "DATE_FORMAT(window_start_time, 'yyyy-MM-dd HH:mm:ss') AS window_start_time      " +
                ", DATE_FORMAT(window_end_time, 'yyyy-MM-dd HH:mm:ss') AS window_end_time      " +
                ", open_ad_id      " +
                ", open_ad_count      " +
                ", open_ad_ms      " +
                ", open_ad_skip_ms      " +
                ", ts FROM join_table"
        );
    }


    private static void createOutputTable(TableEnvironment tabEnv) {
        //app启动类型汇总表
        tabEnv.executeSql("CREATE TABLE dws_traffic_app_start_window_report_clickhouse_sink (   " +
                "`window_start_time` STRING COMMENT '窗口开始日期时间',   " +
                "`window_end_time` STRING COMMENT '窗口结束日期时间',   " +
                "`entry` STRING COMMENT '启动app方式',   " +
                "`entry_count` BIGINT COMMENT '启动app方式次数',   " +
                "`loading_time` BIGINT COMMENT '启动app方式次数',   " +
                "`ts` BIGINT COMMENT '启动app加载时间' " +
                ") WITH (   " +
                "'connector' = 'clickhouse',   " +
                "'url' = 'jdbc:clickhouse://node103:8123/jtp_log_report',   " +
                "'table' = 'dws_traffic_app_start_window_report',   " +
                "'username' = 'default',   " +
                "'password' = '',   " +
                "'format' = 'json' )"
        );

        //app启动广告汇总表
        tabEnv.executeSql("CREATE TABLE dws_traffic_app_start_ad_window_report_clickhouse_sink (   " +
                "`window_start_time` STRING COMMENT '窗口开始日期时间',   " +
                "`window_end_time` STRING COMMENT '窗口结束日期时间',   " +
                "`open_ad_id` STRING COMMENT '启动app方式',   " +
                "`open_ad_count` BIGINT COMMENT '广告展示次数',   " +
                "`open_ad_ms` BIGINT COMMENT '广告总共播放时间',   " +
                "`open_ad_skip_ms` BIGINT COMMENT '用户跳过广告时间',   " +
                "`ts` BIGINT COMMENT '数据计算时间戳' " +
                ") WITH (   " +
                "'connector' = 'clickhouse',   " +
                "'url' = 'jdbc:clickhouse://node103:8123/jtp_log_report',   " +
                "'table' = 'dws_traffic_app_start_ad_window_report',   " +
                "'username' = 'default',   " +
                "'password' = '',   " +
                "'format' = 'json' )"
        );
    }

    private static Table handle1(TableEnvironment tabEnv) {
        //提取启动日志
        Table startLogTable = tabEnv.sqlQuery("SELECT\n" +
                " `start`['entry'] AS entry\n" +
                ", CAST(`start`['loading_time'] AS BIGINT) AS loading_time\n" +
                ", `start`['open_ad_id'] AS open_ad_id\n" +
                ", CAST(`start`['open_ad_ms'] AS BIGINT) AS open_ad_ms\n" +
                ", CAST(`start`['open_ad_skip_ms'] AS BIGINT) AS open_ad_skip_ms\n" +
                ", row_time\n" +
                " FROM dwd_traffic_start_log_kafka_source\n" +
                " WHERE `start`['entry'] IS NOT NULL"
        );

        tabEnv.createTemporaryView("start_log_table", startLogTable);

        //设置窗口 分组 聚合
//        Table startReportTable = tabEnv.sqlQuery("SELECT     " +
//                "TUMBLE_START(row_time, INTERVAL '10' second) AS window_start_time     " +
//                ", TUMBLE_END(row_time, INTERVAL '10' second) AS window_end_time     " +
//                ", entry     " +
//                ", count(entry) AS entry_count     " +
//                ", sum(loading_time) AS loading_time     " +
//                ", UNIX_TIMESTAMP() * 1000 AS ts FROM start_log_table GROUP BY     " +
//                "TUMBLE(row_time, INTERVAL '10' second ),     entry"
//        );

        Table startReportTable = tabEnv.sqlQuery("SELECT\n" +
                " window_start AS window_start_time,\n" +
                " window_end AS window_end_time,\n" +
                " entry,\n" +
                " count(entry) AS entry_count,\n" +
                " sum(loading_time) AS loading_time,\n" +
                " UNIX_TIMESTAMP() * 1000 AS ts\n" +
                " FROM TABLE(\n" +
                " TUMBLE(DATA => TABLE start_log_table,TIMECOL => DESCRIPTOR(row_time),SIZE => INTERVAL '10' second\n" +
                " )\n" +
                ")\n" +
                "GROUP BY window_start ,window_end,entry"
        );


//        Table startAdReportTable = tabEnv.sqlQuery("SELECT     " +
//                "TUMBLE_START(row_time, INTERVAL '10' second) AS window_start_time      " +
//                ", TUMBLE_END(row_time, INTERVAL '10' second) AS window_end_time      " +
//                ", open_ad_id      " +
//                ", count(open_ad_id) AS open_ad_count      " +
//                ", sum(open_ad_ms) AS open_ad_ms      " +
//                ", sum(open_ad_skip_ms) AS open_ad_skip_ms      " +
//                ", UNIX_TIMESTAMP() * 1000 AS ts FROM start_log_table GROUP BY     " +
//                "TUMBLE(row_time, INTERVAL '10' second ),     open_ad_id"
//        );
//        Table startAdReportTable1 = tabEnv.sqlQuery("SELECT" +
//                " window_start AS window_start_time,\n" +
//                " window_end AS window_end_time,\n" +
//                " open_ad_id,\n" +
//                " count(open_ad_id) AS open_ad_count,\n" +
//                " sum(open_ad_ms) AS open_ad_ms,\n" +
//                " sum(open_ad_skip_ms) AS open_ad_skip_ms,\n" +
//                " UNIX_TIMESTAMP() * 1000 AS ts," +
//                " PROCTIME() AS proc_time\n" +
//                " FROM TABLE (\n" +
//                " TUMBLE(TABLE start_log_table,DESCRIPTOR(row_time),INTERVAL '10' second)\n" +
//                " )t1\n" +
//                " GROUP BY\n" +
//                " window_start,window_end,open_ad_id"
//        );
//
//        tabEnv.createTemporaryView("start_ad_report_table", startAdReportTable1);


        return startReportTable;
    }





    private static void createInputTable(TableEnvironment tabEnv) {
        tabEnv.executeSql("CREATE TABLE dwd_traffic_start_log_kafka_source\n" +
                "(\n   " +
                "`common` MAP<STRING, STRING> COMMENT '公共环境信息',\n" +
                "`start` MAP<STRING, STRING> COMMENT '启动信息',\n" +
                "`ts` BIGINT,\n" +
                "row_time AS TO_TIMESTAMP(FROM_UNIXTIME(ts / 1000, 'yyyy-MM-dd HH:mm:ss.SSS')),\n" +
                "WATERMARK FOR row_time AS row_time - INTERVAL '0' MINUTE\n" +
                ") WITH (\n" +
                "'connector' = 'kafka',\n" +
                "'topic' = 'dwd-log-startT-log',\n" +
                "'properties.bootstrap.servers' = 'node101:9092,node102:9092,node103:9092',\n" +
                "'properties.group.id' = 'gid_dws_log_app_start',\n" +
                "'scan.startup.mode' = 'earliest-offset',\n" +
                "'format' = 'json',\n" +
                "'json.fail-on-missing-field' = 'false',\n" +
                "'json.ignore-parse-errors' = 'true'\n" +
                ")"
        );

    }

    private static TableEnvironment getTableEnv() {
        // 环境属性
        EnvironmentSettings settings = EnvironmentSettings.newInstance()
                .inStreamingMode()
                .useBlinkPlanner()
                .build();
        TableEnvironment tabEnv = TableEnvironment.create(settings) ;
        // 配置属性
        Configuration configuration = tabEnv.getConfig().getConfiguration();
        configuration.setString("table.local-time-zone", "Asia/Shanghai");
        configuration.setString("table.exec.resource.default-parallelism", "1");
        configuration.setString("table.exec.state.ttl", "5 s");
        // 返回对象
        return tabEnv;
    }
}
